from loguru import logger
from datetime import datetime
import sys, os
from config import cfg


def log_init(filename, level="INFO"):
    retention_ = "1 days"
    encoding_ = "utf-8"
    logger.remove()
    # 控制台只显示指定level级别的日志信息
    logger.add(sink=sys.stderr, level=level)

    # 设置日志保存目录
    log_path = os.path.join(cfg.log_dir, filename + '_' + str(datetime.now())[:10] + '.txt')
    # 所有Debug级别以上的日志信息全部写入文件
    logger.add(sink=log_path, level="DEBUG", rotation=retention_, encoding=encoding_)
